# autor: Fabiano A. Lima
# Esse arquivo contém o código usado para gera ras tabelas e estimar os modelos.

library(tidyverse)
library(lubridate)
library(stargazer)
library(kableExtra)

source("pesquisador.r", encoding = "utf8")
source("gera_tabela.r", encoding = "utf8")

folha <- folha() 
o_globo <- o_globo()

# Baixando dados folha

#testes dos critérios de busca
data_inicio <- "29/07/2012"
data_fim <- "15/03/2014"


tbl_citacoes_jb <- readRDS("./data/citacoes_jb_mensalao.rds")
tbl_citacoes_jb$Total <- tbl_citacoes_jb$Folha + tbl_citacoes_jb$`O Globo`
gera_tabela_totais(tbl_citacoes_jb)



tbl_citacoes_rl <- readRDS("./data/citacoes_rl_mensalao.rds")
tbl_citacoes_rl$Total <- tbl_citacoes_rl$Folha + tbl_citacoes_rl$`O Globo`
gera_tabela_totais(tbl_citacoes_rl)


df_base <- read_csv("./data/mensalão joaquim barbosa lewandowski 2012-07-29 a 2014-03-15.csv", col_types = "Dddd", na = "<1", skip = 2)

# a regressão com a normalização independente deu o mesmo coeficiente para JB e RL. A diferença entre os dois se dá
# com o dado normalizado.
#df_base <-  read_csv("./data/mensalão 2012-07-29 a 2014-03-15.csv", col_types = "Dd", na = "<1", skip = 2) %>%
#  inner_join(read_csv("./data/joaquim barbosa 2012-07-29 a 2014-03-15.csv", col_types = "Dd", na = "<1", skip = 2), by = "Semana") %>%
#  inner_join(read_csv("./data/lewandowski 2012-07-29 a 2014-03-15.csv", col_types = "Dd", na = "<1", skip = 2), by = "Semana")

df_norm <- df_base %>%
  replace_na(list(`Escândalo do mensalão: (Brasil)` = 0.1, `Joaquim Barbosa: (Brasil)` = 0.1, `Ricardo Lewandowski: (Brasil)` = 0.1)) %>%
  left_join(folha$gera_df(readRDS("./data/jb_sem_mensalao_folha.rds"), "folha_jb"), by = "Semana") %>%
  left_join(o_globo$gera_df(readRDS("./data/jb_sem_mensalao_globo.rds"), "globo_jb"), by = "Semana") %>%
  left_join(folha$gera_df(readRDS("./data/rl_sem_mensalao_folha.rds"), "folha_rl"), by = "Semana") %>%
  left_join(o_globo$gera_df(readRDS("./data/rl_sem_mensalao_globo.rds"), "globo_rl"), by = "Semana") %>%
  replace_na(list(folha_jb = 0, globo_jb = 0, folha_rl = 0, globo_rl = 0)) %>%
  mutate(
    cit_indep_jb = folha_jb + globo_jb,
    cit_indep_rl = folha_rl + globo_rl,
    `Ano.2013` = ifelse(year(Semana) == 2013, 1, 0),
    `Ano.2014` = ifelse(year(Semana) == 2014, 1, 0)
  )


fit_1_jb <- lm(`Joaquim Barbosa: (Brasil)` ~ `Escândalo do mensalão: (Brasil)`, data = df_norm)
fit_2_jb <- lm(`Joaquim Barbosa: (Brasil)` ~ `Escândalo do mensalão: (Brasil)` + cit_indep_jb, data = df_norm)
fit_3_jb <- lm(`Joaquim Barbosa: (Brasil)` ~ `Escândalo do mensalão: (Brasil)` + cit_indep_jb + `Ano.2013` + `Ano.2014`, data = df_norm)

gera_tabela(fit_1_jb, fit_2_jb, fit_3_jb, "2013":"2014", "2012")

fit_1_rl <- lm(`Ricardo Lewandowski: (Brasil)` ~ `Escândalo do mensalão: (Brasil)`, data = df_norm)
fit_2_rl <- lm(`Ricardo Lewandowski: (Brasil)` ~ `Escândalo do mensalão: (Brasil)` + cit_indep_rl, data = df_norm)
fit_3_rl <- lm(`Ricardo Lewandowski: (Brasil)` ~ `Escândalo do mensalão: (Brasil)` + cit_indep_rl + `Ano.2013` + `Ano.2014`, data = df_norm)

gera_tabela(fit_1_rl, fit_2_rl, fit_3_rl, "2013":"2014", "2012")

summary(fit_3_rl)





